package jianzhiOffer;

import java.util.ArrayList;
import java.util.List;

/**
 * 32 - III. 从上到下打印二叉树 III
 * 请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。
 *
 * @author 司松坡
 * @createTime 2023年04月24日
 */
public class Jian32III {
    public List<List<Integer>> levelOrder(TreeNode root) {
        int i = 1;
        List<List<Integer>> res = new ArrayList<>();
        if (root == null) {
            return new ArrayList<>();
        }
        ArrayList<TreeNode> treeNodes = new ArrayList<>();
        treeNodes.add(root);
        min(res, treeNodes,i);
        return res;
    }

    public void min(List<List<Integer>> list, List<TreeNode> treeNodeList, int i) {
        List<TreeNode> newList = new ArrayList<>();
        List<Integer> list1 = new ArrayList<>();
        if (i % 2 == 1) {
            for (int j = 0; j < treeNodeList.size(); j++) {
                TreeNode treeNode = treeNodeList.get(j);
                list1.add(treeNode.val);
                if (treeNode.left != null) {
                    newList.add(treeNode.left);
                }
                if (treeNode.right != null) {
                    newList.add(treeNode.right);
                }
            }
        } else {
            for (int j = treeNodeList.size() - 1; j >= 0; j--) {
                TreeNode treeNode = treeNodeList.get(j);
                list1.add(treeNode.val);
            }
            for (int j = 0; j < treeNodeList.size(); j++) {
                TreeNode treeNode = treeNodeList.get(j);
                if (treeNode.left != null) {
                    newList.add(treeNode.left);
                }
                if (treeNode.right != null) {
                    newList.add(treeNode.right);
                }
            }
        }
        list.add(list1);
        if (null != newList && !newList.isEmpty()) {
            i+=1;
            min(list, newList,i);
        }
    }

}
